package edu.cornell.cs4740.wsd.parsing;

import edu.cornell.cs4740.wsd.Bigram;

public class BigramDistanceContainer {

	Bigram bigram;
	int distance;
	String stemmedWord1;
	String stemmedWord2;
	
	public BigramDistanceContainer(Bigram b, int currentDistance) {
		bigram = b;
		distance = currentDistance;
	}
	public Bigram getBigram() {
		return bigram;
	}
	public void setBigram(Bigram bigram) {
		this.bigram = bigram;
	}
	public String getBigramString() {
	  return bigram.getWord1() + " " + bigram.getWord2();
	}
	public int getDistance() {
		return distance;
	}
	public void setDistance(int ditance) {
		this.distance = ditance;
	}
	
	
	
	public String getStemmedWord1() {
		return stemmedWord1;
	}
	public void setStemmedWord1(String stemmedWord1) {
		this.stemmedWord1 = stemmedWord1;
	}
	public String getStemmedWord2() {
		return stemmedWord2;
	}
	public void setStemmedWord2(String stemmedWord2) {
		this.stemmedWord2 = stemmedWord2;
	}
	public String getStemmedBigramString() {
	  return stemmedWord1 + " " + stemmedWord2;
	}
	@Override
	public String toString() {
		return "Word 1: " + bigram.getWord1() + "\nWord 2: " + bigram.getWord2() + "\nStemmed Word 1: " + stemmedWord1 + "\nStemmed Word 2: " + stemmedWord2 + "\nDistance: " + distance;
		
	}

}
